romantic python code